package com.smartwrite.dto;

import java.util.Map;

/**
 * 报告生成请求参数
 */
public class ReportGenerationRequest {
    
    private String prompt;              // 提示词
    private String model;               // 模型名称
    private Map<String, String> apiKeys; // API密钥映射
    private Boolean enableSearch;       // 是否启用搜索
    private String searchEngine;        // 搜索引擎

    // 构造函数
    public ReportGenerationRequest() {}

    public ReportGenerationRequest(String prompt, String model, Map<String, String> apiKeys) {
        this.prompt = prompt;
        this.model = model;
        this.apiKeys = apiKeys;
        this.enableSearch = true;
        this.searchEngine = "google";
    }

    // Getter和Setter方法
    public String getPrompt() {
        return prompt;
    }

    public void setPrompt(String prompt) {
        this.prompt = prompt;
    }

    public String getModel() {
        return model;
    }

    public void setModel(String model) {
        this.model = model;
    }

    public Map<String, String> getApiKeys() {
        return apiKeys;
    }

    public void setApiKeys(Map<String, String> apiKeys) {
        this.apiKeys = apiKeys;
    }

    public Boolean getEnableSearch() {
        return enableSearch;
    }

    public void setEnableSearch(Boolean enableSearch) {
        this.enableSearch = enableSearch;
    }

    public String getSearchEngine() {
        return searchEngine;
    }

    public void setSearchEngine(String searchEngine) {
        this.searchEngine = searchEngine;
    }
}